package com.cwj.algorithm.graph;

import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * 加权无向图
 * @author chenwujie
 * @date 2020-12-24 09:17
 */
public class EdgeWeightedGraph {
    /**
     * 顶点数量
     */
    private int v;
    /**
     * 边数量
     */
    private int e;
    /**
     * 邻接表
     */
    private LinkedList<Edge>[] adj;
    /**
     * 所有边（不重复）
     */
    private Deque<Edge> edges;

    public EdgeWeightedGraph(int v){
        this.v = v;
        adj = new LinkedList[v];
        for (int i = 0; i < v; i++) {
            adj[i] = new LinkedList<>();
        }
    }

    public int getE() {
        return e;
    }

    public int getV() {
        return v;
    }

    /**
     * 添加一条边
     * @param edge
     */
    public void addEdge(Edge edge){
        adj[edge.getV()].push(edge);
        adj[edge.getW()].push(edge);
        e++;
    }

    public Deque<Edge> edgesByV(int v){
        return adj[v];
    }

    /**
     * 获取所有的边（不重复）
     * @return
     */
    private Deque<Edge> edges(){
        Deque<Edge> edges = new LinkedList<>();
        for (int i = 0; i < v; i++) {
            for (Edge edge : adj[i]) {
                if(edge.getV() == i){
                    edges.push(edge);
                }
            }
        }
        this.edges = edges;
        return edges;
    }

    /**
     * 获取所有的边（不重复）
     * @return
     */
    public List<Edge> getEdges() {
        if(edges == null){
            edges();
        }
        return new LinkedList<>(edges);
    }
}
